from pandas import DataFrame
import pandas as pd

from examples.plugin.common import timer


@timer
def run(param_dict: dict) -> DataFrame:
    df = ""
    if param_dict["type"] == "excel":
        df = pd.read_excel(param_dict["file_name"])
    else:
        df = pd.read_csv(param_dict["file_name"])
    return df
